package com.miguo.framework.authorize;

import com.miguo.framework.app.AppFramework;

import org.json.JSONException;
import org.json.JSONObject;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Sign {

    /**
     * 生成要请求给的参数数组
     *
     * @return 要请求的参数数组
     */
    public static Map<String, String> Encrypt() {
        // 获取默认参数 除去数组中的空值和签名参数
        Map<String, String> params = paraFilter(AppFramework.getDefaultParam());
        // 生成签名结果
        String sign = buildRequestSign(params);
        // 签名结果与签名方式加入请求提交参数组中
        params.put("sign", sign);
        params.put("sign_type", Config.sign_type);
        return params;
    }

    /**
     * 生成要请求给的参数数组
     *
     * @param mapParams 请求前的参数数组
     * @return 要请求的参数数组
     */
    public static Map<String, String> Encrypt(Map<String, String> mapParams) {

        // 获取默认参数
        Map<String, String> default_params = paraFilter(AppFramework.getDefaultParam());

        // 除去数组中的空值和签名参数
        Map<String, String> request_params = paraFilter(mapParams);
        request_params.putAll(default_params);

        // 生成签名结果
        String sign = buildRequestSign(request_params);

        // 签名结果与签名方式加入请求提交参数组中
        request_params.put("sign", sign);
        request_params.put("sign_type", Config.sign_type);
        return request_params;
    }

    /**
     * 生成要请求给的参数数组
     *
     * @param mapParams 请求前的参数数组
     * @return 要请求的参数数组
     */
    public static Map<String, String> EncryptData(Map<String, String> mapParams) {

        // 获取默认参数
        Map<String, String> default_params = paraFilter(AppFramework.getDefaultParam());

        // 除去数组中的空值和签名参数
        Map<String, String> request_params = paraFilter(mapParams);
        request_params.putAll(default_params);

        // 生成签名结果
        String sign = buildRequestSign(request_params);
        String data = RSA.Encrypt(createLinkStringJson(mapParams));

        // 签名结果与签名方式加入请求提交参数组中
        default_params.put("encrypt_data", data);
        default_params.put("sign", sign);
        default_params.put("sign_type", Config.sign_type);
        return default_params;
    }

    /**
     * 除去数组中的空值和签名参数
     *
     * @param params 签名参数组
     * @return 去掉空值与签名参数后的新签名参数组
     */
    private static Map<String, String> paraFilter(Map<String, String> params) {

        Map<String, String> result = new HashMap<String, String>();

        if (params == null || params.size() <= 0) {
            return result;
        }

        for (String key : params.keySet()) {
            String value = params.get(key);
            if (value == null || value.equals("")
                    || key.equalsIgnoreCase("sign")
                    || key.equalsIgnoreCase("sign_type")) {
                continue;
            }
            result.put(key, value);
        }

        return result;
    }

    /**
     * 生成签名结果
     *
     * @param params 要签名的数组
     * @return 签名结果字符串
     */
    private static String buildRequestSign(Map<String, String> params) {
        String prestr = createLinkString(params); // 把数组所有元素，按照“参数=参数值”的模式用“&”字符拼接成字符串
        String sign = "";
        if (Config.sign_type.equals("MD5")) {
            sign = MD5.sign(prestr, Config.input_charset);
        }
        return sign;
    }

    /**
     * 把数组所有元素排序
     *
     * @param params 需要排序并参与字符拼接的参数组
     * @return 拼接后字符串
     */

//	private static Map<String, String> createLinkMap(Map<String, String> params) {
//
//		List<String> keys = new ArrayList<String>(params.keySet());
//		Collections.sort(keys);
//
//		Map<String, String> result = new HashMap<String, String>();
//
//		for (int i = 0; i < keys.size(); i++) {
//			String key = keys.get(i);
//			String value = params.get(key);
//			result.put(key, value);
//		}
//
//		return result;
//	}
    private static String createLinkStringJson(Map<String, String> params) {

        List<String> keys = new ArrayList<String>(params.keySet());
        Collections.sort(keys);

        JSONObject JSON = new JSONObject();
        for (int i = 0; i < keys.size(); i++) {
            String key = keys.get(i);
            String value = params.get(key);

            try {
                JSON.put(key, value);
            } catch (JSONException e) {
                e.printStackTrace();
            }
        }
        return JSON.toString();
    }


    /**
     * 把数组所有元素排序，并按照“参数=参数值”的模式用“&”字符拼接成字符串
     *
     * @param params 需要排序并参与字符拼接的参数组
     * @return 拼接后字符串
     */

    private static String createLinkString(Map<String, String> params) {

        List<String> keys = new ArrayList<String>(params.keySet());
        Collections.sort(keys);

        String prestr = "";

        /*StringBuilder prestr = new StringBuilder();
        for (int i = 0; i < keys.size(); i++) {
            if (prestr.length() > 0) {
                prestr.append("&");
            }
            String key = keys.get(i);
            String value = params.get(key);
            prestr.append(String.format("%s=%s", key, value));
        }*/


        for (int i = 0; i < keys.size(); i++) {
            String key = keys.get(i);
            String value = params.get(key);

            if (i == keys.size() - 1) {// 拼接时，不包括最后一个&字符
                prestr = prestr + key + "=" + value;
            } else {
                prestr = prestr + key + "=" + value + "&";
            }
        }

        return prestr;
    }

}
